package com.carhailing.gateway.filters;

import com.carhailing.common.constants.TokenConstants;
import com.carhailing.common.utils.JwtUtils;
import com.carhailing.common.utils.StringUtils;
import com.carhailing.gateway.config.IgnoreWhiteConfig;
import com.carhailing.gateway.utils.GatewayUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.cloud.gateway.filter.GatewayFilterChain;
import org.springframework.cloud.gateway.filter.GlobalFilter;
import org.springframework.core.Ordered;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.http.server.reactive.ServerHttpRequest;
import org.springframework.stereotype.Component;
import org.springframework.web.server.ServerWebExchange;
import reactor.core.publisher.Mono;

import java.util.List;
import java.util.concurrent.TimeUnit;

/**
 * @ClassName:
 * @Description: 过滤请求，验证 token
 * @Author: zhuwenqiang
 * @Date: 2024/2/28
 */
@Component
public class AuthFilter implements GlobalFilter, Ordered {

    @Autowired
    private IgnoreWhiteConfig ignoreWhitesConfig;

    @Autowired
    private RedisTemplate<String,String> redisTemplate;

    /**
     * 过滤请求方法
     * @param exchange  请求的上下文， 通过这个对象可以获取请求对象以及响应对象
     * @param chain     过滤器链 使用他进行 过滤 或者 放行
     * @return
     */
    @Override
    public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
        // 判断这次请求是否需要验证 token 白名单【直接放行的请求】
        // 获取系统白名单
        List<String> whites = ignoreWhitesConfig.getWhites();
        // 获取当前请求的 URI
        ServerHttpRequest request = exchange.getRequest();
        String path = request.getURI().getPath();
        boolean matches = StringUtils.matches(path, whites);
        if (matches) {
            // 白名单请求
            return chain.filter(exchange);
        }
//        boolean anyMatch = whites.stream().anyMatch(white -> white.equals(path));
//        if (anyMatch) {
        // 白名单请求
//            return chain.filter(exchange);
//        }
        // 验证 token
        String token = request.getHeaders().getFirst(TokenConstants.TOKEN);
        // 非空验证
        if (StringUtils.isBlank(token)) {
            // 空的
            return GatewayUtils.errorResponse(exchange, "token不能为空！");
        }
        // 是否合法
        try {
            JwtUtils.parseToken(token);
        } catch (Exception e) {
            return GatewayUtils.errorResponse(exchange, "token不合法！");
        }
        // 是否有效
        String userKey = JwtUtils.getUserKey(token);
        if (!redisTemplate.hasKey(TokenConstants.LOGIN_TOKEN_KEY + userKey)) {
            return GatewayUtils.errorResponse(exchange, "token过期！");
        } else {
            // 重新设置 redis中的用户有效时间
            redisTemplate.expire(TokenConstants.LOGIN_TOKEN_KEY + userKey, 30, TimeUnit.MINUTES);
        }
        // 放行
        return chain.filter(exchange);
    }

    /**
     * 规定过滤器执行的顺序
     * @return  方法的返回值越小 执行顺序越高
     */
    @Override
    public int getOrder() {
        return 0;
    }

}
